"""
PyMySQL访问数据库 - 插入数据
"""
import pymysql

import common

no = int(input('学号: '))
name = input('姓名: ')
sex = int(input('性别: '))
birth = input('生日: ')
addr = input('家庭住址: ')
col = int(input('学院编号: '))
# 第1步：创建连接
conn = common.get_conn(database='school')
try:
    # 第2步：获取游标
    with conn.cursor() as cursor:
        # 第3步：发出SQL语句获取结果
        result = cursor.execute(
            'insert into tb_student values '
            '(%s, %s, %s, %s, %s, %s)',
            (no, name, sex, birth, addr, col)
        )
    if result == 1:
        print('添加学生成功!!!')
    # 第4步：成功了提交
    conn.commit()
except pymysql.MySQLError as err:
    # 第4步：失败了回滚
    conn.rollback()
    print(err, '添加学生失败!!!')
finally:
    # 第5步：释放连接
    conn.close()
